package com.juejin.www.utils;

import com.juejin.www.constant.Constant;
import io.jsonwebtoken.Claims;
import io.jsonwebtoken.Jwts;
import io.jsonwebtoken.SignatureAlgorithm;

import java.util.Date;
import java.util.Map;

/**
 * JWT令牌工具类
 * @author Jacky
 */
public class JwtUtils {

    private static String signKey = Constant.SIGN_KEY;
    private static Long expire = Constant.EXPIRE;

    /**
     * 生成JWT令牌
     * @param claims JWT第二部分负载payload中存储的内容
     * @return 生成的jwt令牌
     */
    public static String generateJwt(Map<String, Object> claims){
        String jwt = Jwts.builder()
                .addClaims(claims)
                .signWith(SignatureAlgorithm.HS256, signKey)
                .setExpiration(new Date(System.currentTimeMillis() + expire))
                .compact();
        return jwt;
    }

    /**
     * 解析JWT令牌
     * @param jwt JWT令牌
     * @return JWT第二部分负载payload中存储的内容
     */
    public static Claims parseJWT(String jwt){
        Claims claims = Jwts.parser()
                .setSigningKey(signKey)
                .parseClaimsJws(jwt)
                .getBody();
        return claims;
    }

    /**
     * 通过JWT令牌获取用户id
     * @param jwt JWT令牌
     * @return 用户id
     */
    public static Integer getIdByJWT(String jwt){
        Claims claims = parseJWT(jwt);
        return (Integer) claims.get("id");
    }
}
